#!/usr/bin/env python3

# Copyright 2021 Modern Electron

import numpy as np

ref_density = np.array([
    1.29556695e+14, 2.24358819e+14, 2.55381744e+14, 2.55655005e+14,
    2.55796267e+14, 2.55819109e+14, 2.55819687e+14, 2.55751184e+14,
    2.55920806e+14, 2.56072344e+14, 2.55937266e+14, 2.55849080e+14,
    2.55918981e+14, 2.55980835e+14, 2.56054153e+14, 2.56074693e+14,
    2.56036953e+14, 2.56181152e+14, 2.56322617e+14, 2.56253541e+14,
    2.56196222e+14, 2.56353087e+14, 2.56256025e+14, 2.55928999e+14,
    2.56110985e+14, 2.56658916e+14, 2.56832588e+14, 2.56551873e+14,
    2.56491185e+14, 2.56469926e+14, 2.56418623e+14, 2.56541073e+14,
    2.56513772e+14, 2.56424505e+14, 2.56302757e+14, 2.56242393e+14,
    2.56270399e+14, 2.56178952e+14, 2.56071408e+14, 2.56141949e+14,
    2.56419807e+14, 2.56606936e+14, 2.56437775e+14, 2.56252446e+14,
    2.56309518e+14, 2.56383487e+14, 2.56265139e+14, 2.56167672e+14,
    2.56466917e+14, 2.56924869e+14, 2.56901785e+14, 2.56631493e+14,
    2.56643456e+14, 2.56523464e+14, 2.56378270e+14, 2.56571296e+14,
    2.56794304e+14, 2.56788544e+14, 2.56549715e+14, 2.56303160e+14,
    2.56210813e+14, 2.56418356e+14, 2.57314522e+14, 2.58471390e+14,
    2.58169771e+14, 2.56946438e+14, 2.56726546e+14, 2.56853122e+14,
    2.56613699e+14, 2.56509534e+14, 2.56692972e+14, 2.56705133e+14,
    2.56372142e+14, 2.56167556e+14, 2.56296946e+14, 2.56498752e+14,
    2.56523102e+14, 2.56404334e+14, 2.56227096e+14, 2.56398997e+14,
    2.56614907e+14, 2.56436657e+14, 2.56388606e+14, 2.56553679e+14,
    2.56637914e+14, 2.56407785e+14, 2.56104131e+14, 2.56082340e+14,
    2.56095275e+14, 2.56278448e+14, 2.56808134e+14, 2.57127897e+14,
    2.56858174e+14, 2.56326990e+14, 2.56296032e+14, 2.56563349e+14,
    2.56482273e+14, 2.56667481e+14, 2.57072448e+14, 2.56767530e+14,
    2.56433245e+14, 2.56586570e+14, 2.56636412e+14, 2.56765628e+14,
    2.56868130e+14, 2.56783441e+14, 2.56714518e+14, 2.56651014e+14,
    2.56528394e+14, 2.56227520e+14, 2.56163301e+14, 2.56408207e+14,
    2.56433120e+14, 2.56374745e+14, 2.56542028e+14, 2.56748796e+14,
    2.56715201e+14, 2.56298164e+14, 2.56042658e+14, 2.56292455e+14,
    2.56352282e+14, 2.56370562e+14, 2.56487458e+14, 2.56483667e+14,
    2.56741201e+14, 2.56665100e+14, 2.56523784e+14, 2.24741564e+14,
    1.28486944e+14
])

density_data = np.load( 'ion_density_case_1.npy' )
print(repr(density_data))
assert np.allclose(density_data, ref_density)
